package com.daimakuai.tools;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author happyonion.yangtao
 * @Title: GeneratorMain
 * @ProjectName adminb-ehind
 * @PackageName test
 * @Description TODO
 * @date Create at 2018/12/1 17:51
 */
public class GeneratorMain {
    public static void main(String[] args) {
        List<String> warnings = new ArrayList<String>();
        boolean overwrite = true;
        String genConfig = "E:\\CodeFile\\java-stu\\adminb-ehind\\src\\main\\resources\\generatorConfiguration.xml";
        String preFile = GeneratorMain.class.getClassLoader().getResource("").getPath();
        File configFile = new File(genConfig);
        ConfigurationParser cp = new ConfigurationParser(warnings);
        Configuration config = null;

        try {
            config = cp.parseConfiguration(configFile);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (XMLParserException e) {
            e.printStackTrace();
        }

        DefaultShellCallback callback = new DefaultShellCallback(overwrite);
        MyBatisGenerator  myBatisGenerator = null;

        try {
            myBatisGenerator = new MyBatisGenerator(config,callback,warnings);
        } catch (InvalidConfigurationException e) {
            e.printStackTrace();
        }

        try {

            myBatisGenerator.generate(null);
        } catch (SQLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("生成完毕，请刷新查看...");
    }
}
